"""
    导入刚写好的获取数据的类，进行计算
"""

from data_define import Record
from file_define import FileReader, TextFileReader, JsonFileReader
from pyecharts.charts import Bar
from pyecharts.options import *
from pyecharts.globals import ThemeType

# 获取数据
text_data = TextFileReader(
    '/Users/yanfeixingsan/Documents/进阶之路/学习资料/python/第13章资料/2011年1月销售数据.txt')
json_data = JsonFileReader(
    '/Users/yanfeixingsan/Documents/进阶之路/学习资料/python/第13章资料/2011年2月销售数据JSON.txt')
text_list: list[Record] = text_data.read_data()  # 一月数据
json_list: list[Record] = json_data.read_data()  # 二月数据
# 合并数据
all_list = text_list + json_list
data_dict = {}
for record in all_list:
    if record.date in data_dict.keys():
        # 当前日期已经有记录了，所以和老记录做累加即可
        data_dict[record.date] += record.money
    else:
        data_dict[record.date] = record.money

bar = Bar()
bar.add_xaxis(list(data_dict.keys()))
bar.add_yaxis("每日销售额", list(data_dict.values()), label_opts=LabelOpts(is_show=False))
bar.render('每日销售额数据.html')